package com.liuweiheng.cloudproviderpayment8002.util;


import com.liuweiheng.base.utils.ApplicationContextUtil;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ZSetOperations;

import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.TimeUnit;

/**
 * Redis工具类
 *
 * @author lwh
 */
public class RedisUtil {

    private RedisUtil() {}


    @SuppressWarnings("unchecked")
    private static RedisTemplate<String, Object> redisTemplate = (RedisTemplate<String, Object>) ApplicationContextUtil.getBean("redisTemplate");

    public static final long DEFAULT_EXPIRE = 86400L;
    public static final long NOT_EXPIRE = -1L;


    //----------------------------------- String相关

    /**
     * 存储普通对象
     * @param key 键
     * @param value 对象
     */
    public static void setValue(final String key, final Object value) {
        redisTemplate.opsForValue().set(key, value, DEFAULT_EXPIRE, TimeUnit.SECONDS);
    }


    /**
     * 存储普通对象，并设置过期时间（秒）
     * @param key 键
     * @param value 对象
     * @param expire 过期时间（秒）
     */
    public static void setValue(final String key, final Object value, final long expire) {
        redisTemplate.opsForValue().set(key, value, expire, TimeUnit.SECONDS);
    }

    /**
     * 获取普通对象
     * @param key 键
     * @return 对象
     */
    public static Object getValue(final String key) {
        return redisTemplate.opsForValue().get(key);
    }

    /**
     * 删除单个key
     * @param key 键
     * @return true=删除成功 | false=删除失败
     */
    public static boolean delKey(final String key) {
        Boolean b = redisTemplate.delete(key);
        return (b != null) && b;
    }


    /**
     * 删除多个key
     * @param keys key集合
     * @return 成功删除的个数
     */
    public static Long delKey(final Collection<String> keys) {
        Long deleteNum = redisTemplate.delete(keys);
        return deleteNum == null ? 0 : deleteNum;
    }


    //----------------------------------- Hash相关

    /**
     * 存入hash数据
     * @param key 键
     * @param hkey hash键
     * @param value 值
     */
    public static void hashPut(final String key, final String hkey, final Object value) {
        redisTemplate.opsForHash().put(key, hkey, value);
    }


    /**
     * 存入所有hash数据
     * @param key 键
     * @param values hash键值对
     */
    public static void hashPut(final String key, final Map<String, Object> values) {
        redisTemplate.opsForHash().putAll(key, values);
    }


    /**
     * 获取hash数据
     * @param key 键
     * @param hkey hash键
     * @return 对象
     */
    public static Object hashGet(final String key, final String hkey) {
        return redisTemplate.opsForHash().get(key, hkey);
    }

    /**
     * 获取当前键的所有hash数据
     * @param key 键
     * @return Hash对象键值对
     */
    public static Map<Object, Object> hashGetAll(final String key) {
        return redisTemplate.opsForHash().entries(key);
    }


    /**
     * 批量多个hash数据
     * @param key 键
     * @param hkeys hash键
     * @return Hash对象集合
     */
    public static List<Object> hashMultiGet(final String key, final Collection<Object> hkeys) {
        return redisTemplate.opsForHash().multiGet(key, hkeys);
    }

    /**
     * 判断HashKey是否存在
     * @param key 键
     * @param hkey hash键
     * @return true=存在|false=不存在
     */
    public static boolean hasHashKey(final String key, final String hkey) {
        Boolean b = redisTemplate.opsForHash().hasKey(key, hkey);
        return b != null && b;
    }

    /**
     * 批量删除hashkey
     * @param key 键
     * @param keys hash键集合
     * @return 成功删除个数
     */
    public static Long hashDelKey(final String key, final Collection<String> keys) {
        Long deleteNum = redisTemplate.opsForHash().delete(key, keys);
        return deleteNum == null ? 0 : deleteNum;
    }


    //----------------------------------- List相关

    /**
     * 往List中存入数据
     *
     * @param key Redis键
     * @param value 数据
     * @return 存入的个数
     */
    public static long listPush(final String key, final Object value) {
        Long count = redisTemplate.opsForList().rightPush(key, value);
        return count == null ? 0 : count;
    }

    /**
     * 往List中存入多个数据
     *
     * @param key Redis键
     * @param values 多个数据
     * @return 存入的个数
     */
    public static long listPushAll(final String key, final Collection<Object> values) {
        Long count = redisTemplate.opsForList().rightPushAll(key, values);
        return count == null ? 0 : count;
    }


    /**
     * 从List中获取begin到end之间的元素
     *
     * @param key Redis键
     * @param start 开始位置
     * @param end 结束位置（start=0，end=-1表示获取全部元素）
     * @return List对象
     */
    public static List<Object> listGet(final String key, final int start, final int end) {
        return redisTemplate.opsForList().range(key, start, end);
    }

    /**
     * 获取List中所有数据
     * @param key Redis键
     * @return List对象
     */
    public static List<Object> listGetAll(final String key) {
        return redisTemplate.opsForList().range(key, 0, -1);
    }



    //----------------------------------- Set相关

    /**
     * 存储Set结构数据
     * @param key 键
     * @param values 值集合
     * @return 存入的个数
     */
    public static Long setSet(final String key, final Object... values) {
        Long addNum = redisTemplate.opsForSet().add(key, values);
        return addNum == null ? 0 : addNum;
    }

    /**
     * 获取Set结构数据
     * @param key 键
     * @return Set集合
     */
    public static Set<Object> getSetAll(final String key) {
        return redisTemplate.opsForSet().members(key);
    }

    /**
     * 移除Set中的数据
     * @param key 键
     * @param values 需要移除的值集合
     */
    public static Long setDel(final String key, final Object... values) {
        Long removeNum = redisTemplate.opsForSet().remove(key, values);
        return removeNum == null ? 0 : removeNum;
    }


    //----------------------------------- ZSet相关

    /**
     * 往ZSet中存入数据
     *
     * @param key Redis键
     * @param values 值
     * @return 存入的个数
     */
    public static long zsetSet(final String key, final Set<ZSetOperations.TypedTuple<Object>> values) {
        Long count = redisTemplate.opsForZSet().add(key, values);
        return count == null ? 0 : count;
    }

    /**
     * 删除ZSet中的数据
     *
     * @param key Redis键
     * @param values 值
     * @return 移除的个数
     */
    public static long zsetDel(final String key, final Set<ZSetOperations.TypedTuple<Object>> values) {
        Long count = redisTemplate.opsForZSet().remove(key, values);
        return count == null ? 0 : count;
    }


    /**
     * 设置有效时间
     * @param key 键
     * @param expire 过期时间（默认秒）
     * @return true=成功 | false=失败
     */
    public static boolean expire(final String key, final long expire) {
        return expire(key, expire, TimeUnit.SECONDS);
    }

    /**
     * 设置有效时间
     * @param key 键
     * @param expire 过期时间
     * @param unit 时间单位
     * @return true=成功 | false=失败
     */
    public static boolean expire(final String key, final long expire, final TimeUnit unit) {
        Boolean b = redisTemplate.expire(key, expire, unit);
        return (b != null) && b;
    }


}
